package study.core.rdd.operator

import org.apache.spark.rdd.RDD
import org.apache.spark.{SparkConf, SparkContext}

/**
 * 转换算子之groupBy操作
 * 将数据根据指定的规则进行分组, 分区默认不变，但是数据会被打乱重新组合，我们将这样
 * 的操作称之为 shuffle
 *
 * @author zh
 * @date 2021/5/15 10:47
 */
object TestGroupBy {
  def main(args: Array[String]): Unit = {
    val sparkConf = new SparkConf().setMaster("local[*]").setAppName("rddMap")
    val context = new SparkContext(sparkConf)

    // 准备集合
    val list = List(1, 2, 3, 4)
    // 创建RDD
    val rdd: RDD[Int] = context.makeRDD(list, 2)

    // 会根据函数返回值作为key来分组
    val result = rdd.groupBy((num: Int) => num % 2)

    result.collect().foreach(println)

    context.stop()

    //(0,CompactBuffer(2, 4))
    //(1,CompactBuffer(1, 3))

  }

}
